package com.anxuan.agent.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 创建自定义的上下文查询增强器的工厂
 */
public class LoveAppContextualQueryAugmenterFactory {

    public static ContextualQueryAugmenter createInstance(){

        PromptTemplate emptyContextPromptTemplate = new PromptTemplate(
                """
                        你应该输出下面的内容：
                        抱歉，我只能回答你恋爱相关的问题，别的没办法帮到你哦，
                        然后告诉用户如果恋爱之外的问题，可以咨询电话110哦。
                        """
        );

        return ContextualQueryAugmenter.builder()
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
